import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Test;

/**
 * 198. 打家劫舍
 * 你是一个专业的小偷，计划偷窃沿街的房屋。每间房内都藏有一定的现金，影响你偷窃的唯一制约因素就是相邻的房屋装有相互连通的防盗系统，如果两间相邻的房屋在同一晚上被小偷闯入，系统会自动报警。
 *
 * 给定一个代表每个房屋存放金额的非负整数数组，计算你 不触动警报装置的情况下 ，一夜之内能够偷窃到的最高金额。
 */
public class Rob1 {

    @Test
    public void test1() {
        Assertions.assertEquals(4, rob(new int[]{1, 2, 3, 1}));
    }

    @Test
    public void test2() {
        Assertions.assertEquals(12, rob(new int[]{2, 7, 9, 3, 1}));
    }

    @Test
    public void test3() {
        Assertions.assertEquals(4, rob(new int[]{2, 1, 1, 2}));
    }


    public int rob(int[] nums) {

        if (nums.length == 1) {
            return nums[0];
        }
//      打劫到第i家时的总利润
        int[] robProfit = new int[nums.length];
//      选择打劫第0家，那么利润就是第0家的所有资产
        robProfit[0] = nums[0];
//      打劫到第1家，那么总利润要么是第一家的总资产，要么就是第二家的总资产
        robProfit[1] = Math.max(nums[0], nums[1]);

//      从第3家开始继续打劫
        for (int i = 2; i < nums.length; i++) {

            robProfit[i] = Math.max(
//                    不打劫第i家，那么利润就是打劫的第i-1家
                    robProfit[i - 1],
//                    打劫第i家，那么利润就是第i-2家和第i家的利润
                    robProfit[i - 2] + nums[i]
            );
        }


        return robProfit[robProfit.length - 1];
    }
}
